# test
from agent.tools.tools_def import predict_grade_for_stu,predict_grade_for_class
from agent.tools.tools_def import query_database, draw_graph, rag_from_policy
from agent.agent import cot_agent

if __name__ == "__main__":
    # r1 = predict_grade_for_stu("18988889999", "DWDwwdwq")
    # print(r1, type(r1))
    # r2 = predict_grade_for_class("计211", "DWDwwdwq")
    # print(r2, type(r2))
    # r3 = query_database("计211班级的成绩", "姓名, 课程名, 成绩")
    # print(r3, type(r3))
    # r3 = draw_graph("画柱状图", r3)
    # print(r3, type(r3))
    # r5 = rag_from_policy("竞赛经费")
    # print(r5, type(r5))
    # r6 = cot_agent("根据我们班级（计211）的现有成绩，有多少人可以获得奖学金，画出他们的成绩比较")
    # r6 = cot_agent("预测我们班级（计211），谁大学物理会挂科，画出图")
    r6 = cot_agent("预测我的大学物理成绩，会挂科吗。我是张三")
    print(r6)
